package com.javaee.wordtree.service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.javaee.wordtree.entity.ModelChat;
import com.javaee.wordtree.mapper.ModelChatMapper;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.Date;
import java.util.List;

/**
 * 模型聊天服务类
 */
@Service
public class ModelChatService extends ServiceImpl<ModelChatMapper, ModelChat> {

    @Resource
    private ModelChatMapper modelChatMapper;

    /**
     * 获取用户的所有模型聊天记录
     * 
     * @param userId 用户ID
     * @return 聊天列表
     */
    public List<ModelChat> getUserChats(String userId) {
        return modelChatMapper.findChatsByUserId(userId);
    }

    /**
     * 获取用户的特定类型的模型聊天记录
     * 
     * @param userId   用户ID
     * @param chatType 聊天类型
     * @return 聊天列表
     */
    public List<ModelChat> getUserChatsByType(String userId, String chatType) {
        return modelChatMapper.findChatsByUserIdAndType(userId, chatType);
    }

    /**
     * 创建新的模型聊天
     * 
     * @param userId     用户ID
     * @param title      聊天标题
     * @param chatType   聊天类型
     * @param difficulty 难度级别
     * @param keywords   关键词/主题
     * @return 新创建的聊天对象
     */
    public ModelChat createChat(String userId, String title, String chatType, String difficulty, String keywords) {
        ModelChat chat = new ModelChat();
        chat.setUserId(userId);
        chat.setTitle(title);
        chat.setChatType(chatType);
        chat.setDifficulty(difficulty);
        chat.setKeywords(keywords);
        Date now = new Date();
        chat.setCreatedAt(now);
        chat.setUpdatedAt(now);

        save(chat); // 使用MyBatis-Plus提供的save方法保存对象
        return chat;
    }

    /**
     * 更新聊天标题
     * 
     * @param chatId 聊天ID
     * @param title  新标题
     * @return 是否更新成功
     */
    public boolean updateChatTitle(Integer chatId, String title) {
        ModelChat chat = getById(chatId);
        if (chat == null) {
            return false;
        }

        chat.setTitle(title);
        chat.setUpdatedAt(new Date());
        return updateById(chat);
    }

    /**
     * 更新聊天的更新时间
     * 
     * @param chatId 聊天ID
     * @return 是否更新成功
     */
    public boolean updateChatTime(Integer chatId) {
        ModelChat chat = getById(chatId);
        if (chat == null) {
            return false;
        }

        chat.setUpdatedAt(new Date());
        return updateById(chat);
    }

    /**
     * 根据ID获取聊天记录
     * 
     * @param chatId 聊天ID
     * @return 聊天对象
     */
    public ModelChat getChatById(Integer chatId) {
        return getById(chatId); // 使用MyBatis-Plus提供的getById方法
    }

    /**
     * 删除聊天记录
     * 
     * @param chatId 聊天ID
     * @return 是否删除成功
     */
    public boolean deleteChat(Integer chatId) {
        return removeById(chatId); // 使用MyBatis-Plus提供的removeById方法
    }

    /**
     * 检查聊天是否属于指定用户
     * 
     * @param chatId 聊天ID
     * @param userId 用户ID
     * @return 是否属于该用户
     */
    public boolean isChatBelongToUser(Integer chatId, String userId) {
        LambdaQueryWrapper<ModelChat> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(ModelChat::getChatId, chatId).eq(ModelChat::getUserId, userId);
        return count(queryWrapper) > 0;
    }
}